import os

import time
import requests
from lxml import etree

def bizhishe(page):
    """

    :rtype: object
    """
    # 获取源代码

    url = f"https://www.toopic.cn/dnbz/?q=--65--.html/page=&page={page}"
    # 请求头，模拟浏览器
    headers = {
        # 反反爬
        'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/109.0.0.0 Safari/537.36 Edg/109.0.1518.78'
    }
    # r就是源代码
    r = requests.get(url, headers=headers).text
    # print(r)

    # 解析源代码，获取所有图片地址
    ret = etree.HTML(r) #转换成xml类型
    # 获得任意位置img标签
    a_list = ret.xpath('//ul/li/div/a')
    i = 1
    for a in a_list:
        img = a.xpath('.//img/@data-original')[0]
        r = requests.get('https://www.toopic.cn' + img).content  # 图片的二进制数字
        # 保存图片
        with open(f'壁纸社/{page}-{i}.jpg', 'wb') as f:
            f.write(r)
            print(img)
            i = i + 1
            time.sleep(1)

if __name__ == '__main__':
    # 如果当前文件夹没有'壁纸社'文件夹
    if not os.path.exists('壁纸社'):
         # 则创建'壁纸社'文件夹
        os.mkdir('壁纸社')
        for page in range(1, 5):
            print(page)
    bizhishe(page)